Skip to content

feat!: Yolo object detection#987

Open
benITo47 wants to merge 9 commits intomainfrom
@bo/yoloObjectDetection
Open

feat!: Yolo object detection#987
benITo47 wants to merge 9 commits intomainfrom
@bo/yoloObjectDetection

Conversation

@benITo47
Copy link
Contributor

@benITo47 benITo47 commented Mar 20, 2026

Description

This PR adds yolo models and multi-method support to our ObjectDetection API.
Deduplication of common CV code will be done in a separate PR

Introduces a breaking change?

  • Yes
  • No

Type of change

  • Bug fix (change which fixes an issue)
  • New feature (change which adds functionality)
  • Documentation update (improves or adds clarity to existing documentation)
  • Other (chores, tests, code style improvements etc.)

Tested on

  • iOS
  • Android

Testing instructions

Try running example apps with Yolo models
Run tests

Screenshots

Related issues

Closes #718

Checklist

  • I have performed a self-review of my code
  • I have commented my code, particularly in hard-to-understand areas
  • I have updated the documentation accordingly
  • My changes generate no new warnings

Additional notes

@msluszniak msluszniak added the feature PRs that implement a new feature label Mar 20, 2026
@benITo47 benITo47 force-pushed the @bo/yoloObjectDetection branch from aec863e to 9b77da7 Compare March 20, 2026 13:46
@msluszniak msluszniak self-assigned this Mar 23, 2026
@msluszniak msluszniak changed the title @bo/yolo object detection feat!: Yolo object detection Mar 23, 2026
Comment on lines +136 to +140
* @example
* ```typescript
* const sizes = model.getAvailableInputSizes(); // [384, 512, 640] for YOLO models, or undefined for RF-DETR
* ```
*/
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I'm wondering if we should keep this example here, no other typedocs have example section

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

What do you think cc: @NorbertKlockiewicz @chmjkb ?

Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I think the most important is to keep it consistent between modules, so I would prefer to remove it.

Comment on lines +215 to +221
* @example
* ```typescript
* const detections = await model.forward('path/to/image.jpg', {
* detectionThreshold: 0.7,
* inputSize: 640, // For YOLO models
* classesOfInterest: ['PERSON', 'CAR'],
* });
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Same here

Comment on lines +37 to +42
| RF_DETR_NANO (XNNPACK FP32) | TBD | TBD |
| YOLO26N (XNNPACK FP32) | TBD | TBD |
| YOLO26S (XNNPACK FP32) | TBD | TBD |
| YOLO26M (XNNPACK FP32) | TBD | TBD |
| YOLO26L (XNNPACK FP32) | TBD | TBD |
| YOLO26X (XNNPACK FP32) | TBD | TBD |
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

TODO: Add this times

Comment on lines +55 to +60
| RF_DETR_NANO (XNNPACK FP32) | TBD | TBD |
| YOLO26N (XNNPACK FP32) | TBD | TBD |
| YOLO26S (XNNPACK FP32) | TBD | TBD |
| YOLO26M (XNNPACK FP32) | TBD | TBD |
| YOLO26L (XNNPACK FP32) | TBD | TBD |
| YOLO26X (XNNPACK FP32) | TBD | TBD |
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

TODO: Add times here

Co-authored-by: Mateusz Sluszniak <56299341+msluszniak@users.noreply.github.com>
@msluszniak msluszniak marked this pull request as ready for review March 23, 2026 10:04
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

feature PRs that implement a new feature

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Add YOLO support

3 participants